# tests/base.py
import unittest
from init_app import create_app, sqldb

class BaseTestCase(unittest.TestCase):
    """所有测试用例的基类"""
    
    def setUp(self):
        self.app = create_app('test') # 创建测试应用
        self.app_context = self.app.app_context()
        self.app_context.push()
        self.client = self.app.test_client()
        
        # 创建数据库表
        sqldb.create_all()
    
    def tearDown(self):
        sqldb.session.remove() # 关闭数据库连接
        # sqldb.drop_all() # 删除数据库表
        self.app_context.pop()